CUDA_VISIBLE_DEVICES=4,5 python test_gpu.py -tgpu